#!/usr/bin/env python3
# -*- coding: UTF-8 -*-
###########################################################################
# Copyright © 1998 - 2025 Tencent. All Rights Reserved.
###########################################################################
"""
Author: Tencent AI Arena Authors
"""


# Configuration, including dimension settings, algorithm parameter settings.
# The last few configurations in the file are for the Kaiwu platform to use and should not be changed.
# 配置，包含维度设置，算法参数设置，文件的最后一些配置是开悟平台使用不要改动
class Config:

    # features
    # 特征
    FEATURES = [
        2,
        6,
        6,
        16,  # legal_action 16维
        3,  # 超级闪现特征
    ]
    FEATURE_SPLIT_SHAPE = FEATURES
    DIM_OF_OBSERVATION = sum(FEATURES)
    DIM_OF_ACTION_DIRECTION = 16
    DIM_OF_TALENT = 0
    DIM_OF_SUPER_FLASH = 0
    SAMPLE_DIM = 2 * (DIM_OF_OBSERVATION + DIM_OF_ACTION_DIRECTION) + 4
    TARGET_UPDATE_FREQ = 200
    GAMMA = 0.9
    EPSILON_MIN = 0.05  # 从0.1改为0.05
    EPSILON_MAX = 1.0
    EPSILON_DECAY = 1e-7  # 从5e-7改为1e-7，衰减更慢
    START_LR = 2e-4  # 学习率提升
